package com.funny.stock.base.util;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.io.InputStreamReader;
import java.io.LineNumberReader;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Enumeration;

import javax.servlet.http.HttpServletRequest;

public class IPUtils {
    private static final Logger logger = LoggerFactory.getLogger(IPUtils.class);
    private IPUtils() {
    }

    /**
     * 获取当前网络ip
     * @param request
     * @return
     */
    public static String getIpAddr(HttpServletRequest request){
        String ipAddress = request.getHeader("x-forwarded-for");
        if(isUnknownAddress(ipAddress)) {
            ipAddress = request.getHeader("Proxy-Client-IP");
        }
        if(isUnknownAddress(ipAddress)) {
            ipAddress = request.getHeader("WL-Proxy-Client-IP");
        }
//        if(isUnknownAddress(ipAddress)) {
//            ipAddress = request.getHeader("x-real-ip");
//        }
        if(isUnknownAddress(ipAddress)) {
            ipAddress = request.getRemoteAddr();
            if(isLocalAddress(ipAddress)){
                //根据网卡取本机配置的IP
                InetAddress inet = null;
                try {
                    inet = InetAddress.getLocalHost();
                    ipAddress = inet.getHostAddress();
                } catch (UnknownHostException e) {
//                    e.printStackTrace();
                }
            }
        }
        //对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照','分割
        if(ipAddress != null && ipAddress.length() > 15){ //"***.***.***.***".length() = 15
            String[] split = ipAddress.split(",");
            for (String item : split) {
                if(!isLocalAddress(item)){
                    ipAddress = item;
                    break;
                }
            }
//            if(ipAddress.indexOf(",") > 0){
//                ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));
//            }
        }

        //本地ip地址打印 request头信息
        if(ipAddress != null && isLocalAddress(ipAddress)){
            Enumeration<String> names = request.getHeaderNames();
            StringBuilder sb = new StringBuilder();
            while (names.hasMoreElements()){
                String key = names.nextElement();
                sb.append(key);
                sb.append(":");
                sb.append(request.getHeader(key));
                sb.append(";");
            }
            logger.info("request headers:{}",sb.toString());
        }
        return ipAddress;
    }

    private static boolean isUnknownAddress(String ipAddress){
        return ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress);
    }

    private static boolean isLocalAddress(String ipAddress){
        return ipAddress.equals("127.0.0.1") || ipAddress.equals("0:0:0:0:0:0:0:1");
    }

    /**
     * 获得MAC地址
     * @param ip
     * @return
     */
    public static String getMACAddress(String ip){
        String str = "";
        String macAddress = "";
        try {
            Process p = Runtime.getRuntime().exec("nbtstat -A " + ip);
            InputStreamReader ir = new InputStreamReader(p.getInputStream());
            LineNumberReader input = new LineNumberReader(ir);
            for (int i = 1; i < 100; i++) {
                str = input.readLine();
                if (str != null) {
                    if (str.indexOf("MAC Address") > 1) {
                        macAddress = str.substring(str.indexOf("MAC Address") + 14, str.length());
                        break;
                    }
                }
            }
        } catch (IOException e) {
            e.printStackTrace(System.out);
        }
        return macAddress;
    }

}